import numpy as np
import matplotlib.pyplot as plt

seed = 2
def generateds():
    rdm = np.random.RandomState(seed)
    # 用随机数生成数据集
    X = rdm.randn(300,2)
    # 标签制作
    Y_ = [int(x0*x0 + x1*x1<2) for (x0,x1) in X]
    # 遍历Y，1赋值为red，其余blue
    Y_c = [['red' if y else 'blue'] for y in Y_]
    # shape整理， 第一个元素为-1表示随第二个参数计算得到，第二个元素表示多少列，把X
    # 整理为n行2列，把Y整理为n行1列
    X = np.vstack(X).reshape(-1,2)
    Y_ = np.vstack(Y_).reshape(-1,1)

    return X, Y_, Y_c